package com.wangyi.org.service.impl;

import com.wangyi.basic.service.impl.BaseServiceImpl;
import com.wangyi.org.domain.Department;
import com.wangyi.org.mapper.DepartmentMapper;
import com.wangyi.org.service.IDepartmentService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;


/**
 * 业务实现类：部门
 */
@Service
public class DepartmentServiceImpl extends BaseServiceImpl<Department> implements IDepartmentService {

    @Autowired
    private DepartmentMapper departmentMapper;

    @Transactional
    @Override
    public void add(Department department) {
        // 添加之后会返回一个自增长id
        departmentMapper.save(department);
        // 添加的时候对dirPath的处理
        String dirPath = "";
        if(department.getParent_id() == null) { //  没有选择上级部门 or // 选了之后×掉了
            Long id  = department.getId();
            dirPath = "/" + id;
        }else{
            // 获取上级部门
            Long parent_id = department.getParent_id();
            // 获取上级部门的dirPath
            String parentDirPath = departmentMapper.loadById(parent_id).getDirPath();
            dirPath = parentDirPath + "/" + department.getId();
        }

        department.setDirPath(dirPath);
        departmentMapper.update(department);
    }

    @Transactional
    @Override
    public void update(Department department) {
        // 修改的时候对dirPath的处理
        String dirPath = "";
        if(department.getParent_id() == null) { //  没有选择上级部门 or // 选了之后×掉了
            Long id  = department.getId();
            dirPath = "/" + id;
        }else{
            // 获取上级部门
            Long parent_id = department.getParent_id();
            // 获取上级部门的dirPath
            String parentDirPath = departmentMapper.loadById(parent_id).getDirPath();
            dirPath = parentDirPath + "/" + department.getId();
        }
        // 设置新的dirPath
        department.setDirPath(dirPath);
        departmentMapper.update(department);
    }

    @Override
    public List<Department> deptTree() {
//         1. 一次性查询出所有数据，放到内存中，可以使用map
//         2. 从所有数据中找出第一层级/顶级部门：这就是我们需要返回的数据
//         3. 遍历所有部门只要不是第一层级的，找自己的上级，并将自己放入上级的children中
        // 将所有的部门撞到map中  - 每次获取父部门都从这里获取
        List<Department> all = departmentMapper.loadAll();
        Map<Long,Department> map  = new HashMap<>();
        for (Department department : all) {
            map.put(department.getId(),department);
        }

        // 前端需要得到的集合数据 - 部门树
        List<Department> deptTree = new ArrayList<>();
        for (Department department : all) {
            if (department.getParent_id() == null) { //  顶级部门
                deptTree.add(department);
            } else { //  不是顶级部门
                // 1.想办法找到当前部门的 父部门
                Long parent_id = department.getParent_id();
                // 这种从数据库查询的方式，每次通过id去查询 - 查询出来映射给实体domain，children集合中都会是空集合 - 没有任何数据
                // Department parentDepartment = departmentMapper.loadById(parent_id);
                // 下面这种是从Map中去获取的 - 以前装进去的对象还是会存在
                Department parentDepartment = map.get(parent_id);
                // 2.将自己装到  父部门 的children中
                parentDepartment.getChildren().add(department);
            }
        }
        return deptTree;
    }
}
